import sqlite3
import os

# 数据库文件路径
db_path = r'd:\manage1\graduation-project\manage1\manage1\manage\db.sqlite3'

try:
    # 连接到SQLite数据库
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()
    
    print("开始重新创建用户管理表...")
    
    # 删除现有的表（如果存在）
    cursor.execute("DROP TABLE IF EXISTS property_user")
    print("已删除现有表（如果存在）")
    
    # 创建新的用户管理表（不含时间字段）
    cursor.execute('''
    CREATE TABLE property_user (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        user_type TEXT NOT NULL,
        username TEXT NOT NULL UNIQUE,
        password TEXT NOT NULL,
        status TEXT NOT NULL
    )''')
    print("✓ 已创建新的用户管理表，不含时间字段")
    
    # 插入示例数据
    try:
        # 插入管理员用户
        cursor.execute("INSERT INTO property_user (user_type, username, password, status) VALUES (?, ?, ?, ?)", 
                      ('管理员', 'admin', 'admin123', '激活'))
        # 插入普通用户
        cursor.execute("INSERT INTO property_user (user_type, username, password, status) VALUES (?, ?, ?, ?)", 
                      ('普通用户', 'user1', 'user123', '激活'))
        conn.commit()
        print("✓ 已插入示例数据")
    except sqlite3.IntegrityError as e:
        print(f"插入数据时出错: {e}")
    
    # 验证表结构
    print("\n验证新表结构:")
    cursor.execute("PRAGMA table_info(property_user)")
    columns = cursor.fetchall()
    for col in columns:
        print(f"  - {col[1]}: {col[2]}")
    
    # 显示插入的数据
    print("\n验证插入的数据:")
    cursor.execute("SELECT id, user_type, username, status FROM property_user")
    rows = cursor.fetchall()
    print(f"  共 {len(rows)} 条记录:")
    for row in rows:
        print(f"  - ID: {row[0]}, 类型: {row[1]}, 用户名: {row[2]}, 状态: {row[3]}")
    
    print("\n✓ 用户管理表重建完成，已移除创建时间和更新时间字段")
    
except sqlite3.Error as e:
    print(f"数据库错误: {e}")
except Exception as e:
    print(f"发生错误: {e}")
finally:
    if conn:
        conn.close()
        print("\n数据库连接已关闭")
